Fill

使用给定的常量值填充输出数组。该算子将输入的 value 按指定数据类型复制，并在多核环境下利用 DMA 高效地完成填充操作。

\[dst_i = value \quad \text{for} \quad i = 0,1,\dots,N-1\]
输入：

  • value - 待填充的常量值地址。

  • param - Fill 参数结构体指针，描述输出张量的形状与数据类型信息。

  • core_mask - 核掩码（仅适用于共享存储版本）。

输出：

  • output - 填充结果地址。

FillParameter 说明：

FillParameter 用于描述 Fill 算子的输出张量信息，其定义如下：

typedef struct FillParameter {
    int* shape_;      // 张量各维度大小
    int ndim_;        // 张量维度数
    int elem_cnt_;    // 张量元素总数
    int type_size_;   // 单个元素字节大小
} FillParameter;
支持平台：

FT78NE MT7004

备注

  • FT78NE 支持fp, dp, int8, int16, int32, cplx64, cplx128

  • MT7004 支持hp, fp, int16, int32, cplx64

共享存储版本:

void i8_fill_s(int8_t *value, int8_t *output, int core_mask, FillParameter *param)
void i16_fill_s(int16_t *value, int16_t *output, int core_mask, FillParameter *param)
void i32_fill_s(int32_t *value, int32_t *output, int core_mask, FillParameter *param)
void hp_fill_s(half *value, half *output, int core_mask, FillParameter *param)
void fp_fill_s(float *value, float *output, int core_mask, FillParameter *param)
void dp_fill_s(double *value, double *output, int core_mask, FillParameter *param)
void c64_fill_s(float *value, float *output, int core_mask, FillParameter *param)
void c128_fill_s(double *value, double *output, int core_mask, FillParameter *param)

C调用示例：

 1//FT78NE示例
 2#include <stdio.h>
 3#include <fill.h>
 4
 5int main(int argc, char* argv[]) {
 6    float value = 1.0f;
 7    float *output = (float *)0xC0000000;
 8    FillParameter param;
 9    param.elem_cnt_ = 1024;
10    param.type_size_ = sizeof(float);
11    int core_mask = 0xff;
12    fp_fill_s(&value, output, core_mask, &param);
13    return 0;
14}

私有存储版本:

void i8_fill_p(int8_t *value, int8_t *output, FillParameter *param)
void i16_fill_p(int16_t *value, int16_t *output, FillParameter *param)
void i32_fill_p(int32_t *value, int32_t *output, FillParameter *param)
void hp_fill_p(half *value, half *output, FillParameter *param)
void fp_fill_p(float *value, float *output, FillParameter *param)
void dp_fill_p(double *value, double *output, FillParameter *param)
void c64_fill_p(float *value, float *output, FillParameter *param)
void c128_fill_p(double *value, double *output, FillParameter *param)

C调用示例：

 1//FT78NE示例
 2#include <stdio.h>
 3#include <fill.h>
 4
 5int main(int argc, char* argv[]) {
 6    float value = 1.0f;
 7    float *output = (float *)0x10820000;
 8    FillParameter param;
 9    param.elem_cnt_ = 1024;
10    param.type_size_ = sizeof(float);
11    fp_fill_p(&value, output, &param);
12    return 0;
13}